torch
tqdm
ray
numpy
torchvision
tensorboard
wandb